from classeCliente import *
from Repositorio_Cliente import *
from classeCarro import *

from datetime import date
#CREATE TABLE locacao idcli INTEGER, nome VARCHAR, coddvd INTEGER, titulo VARCHAR, codlocacao INTEGER, datalocacao VARCHAR, datadevolucao VARCHAR
#CRIADO POR EDUARDO JOSE
class Locadora:
    try:
        def locar(self):
            import MySQLdb
            db = MySQLdb.connect('localhost','root','root')
            db.select_db('locadora')
            cursor = db.cursor()
            cpf = raw_input("CPF:\n")
            sql = ('SELECT * FROM cliente WHERE cpf = "%s"'%(cpf))
            cursor.execute(sql)
            rs = cursor.fetchall()
            a = 0
            if(rs==()):
                print 'CLIENTE NAO ENCONTRADO'
                a = 1
            else:
                for i in rs:                    
                    cpf = i[0]
                    nome = i[1]
                    endereco = i[2]
                    print 'CPF: %s | NOME: %s | ENDERECO: %s '%(cpf,nome,endereco)
                
                
                
                if(a==0):
                    placa = raw_input("PLACA: \n")
                    sql1 = 'SELECT * FROM carro WHERE placa = "%s"'%(placa)
                    cursor.execute(sql1)
                    rs1=cursor.fetchall()
                    b=0
                    if(rs1 == ()):
                        print 'DVD NAO LOCALIZADO'
                        b = 1

                    else:
                        for i in rs1:
                            marca = i[3]
                            cor = i[4]
                            placa = i[1]
                            modelo = i[2]
                            ano = i[5]
                            print 'MARCA: %s | MODELO: %s | ANO: %s | COR: %s | PLACA: %s'%(marca,modelo, ano,cor,placa)
                        
                        b = 0

                if(b==0):
                    
                    confirm = raw_input("CONFIRMA ESSES DADOS?(S/N)\n").upper()
                    if(confirm == "S"):                       
                            import datetime
                            datalocacao = datetime.date.today()
                            datadevolucao = (datalocacao + datetime.timedelta(days=15)).strftime('%Y-%m-%d')
                            
                            
                            sql3 = 'INSERT INTO locacao (cpf,nome,placa,dataInicio,dataFim) VALUES ("%s","%s","%s","%s","%s")'%(cpf,nome,placa,datalocacao,datadevolucao)
                            cursor.execute(sql3)
                            
                            db.commit()
                            db.close()
                            print 'LOCACAO EFETUADA COM SUCESSO'
                            print 'DEVOLUCAO PREVISTA PARA %s' %(datadevolucao)
                    else:
                        print 'LOCACAO CANCELADA'

        def devolver(self):
                import MySQLdb
                db = MySQLdb.connect('localhost','root','root')
                db.select_db('locadora')
                cursor = db.cursor()
                codigo = raw_input("CODIGO LOCACAO:\n")
                e=0
                #FUNCAO PARA PEGAR A QUANTIDADE E INCREMENTAR QUANDO DEVOLVER
                sql1 = 'SELECT * FROM locacao WHERE idLocacao = "%s"'%(codigo)
                cursor.execute(sql1)
                rs1=cursor.fetchall()
                b=0
                if(rs1 == ()):
                    print 'CODIGO INVALIDO'
                    b = 1
                else:
                    for i in rs1:
                        codigo = i[0]
                        placa = i[3]
                        devolucao = i[4]
                        cpf = i[1]
                        nome = i[2]
                #####################  
                
                print 'CPF: %s\nNOME CLIENTE: %s\nPLACA: %s\nDATA PREVISTA PARA DEVOLUCAO: %s'%(cpf,nome,placa,devolucao)
                confirmacao = raw_input('CONFIRMA OS DADOS ACIMA(S/N)?').upper()
                if(confirmacao == "S"):                    
                    
                    sql4 = 'DELETE FROM locacao WHERE idLocacao = "%s"'%(codigo)
                    
                    cursor.execute(sql4)
                    db.commit()
                    db.close()
                    print 'DEVOLUCAO EFETUADA COM SUCESSO'
                else:
                    print 'OPERACAO CANCELADA'
                        
                        
                       
    except:
        print "ERRO NAO ESPERADO"
                    
##locadora = Locadora()
##locadora.locar()
            
                
